import numpy as np # linear algebra
import pandas as pd # data processing, CSV file I/O
import matplotlib.pyplot as plt
import week5_drawFigure
import week5_FEtrain_lr
import week5_FELog_lr

train = pd.read_csv("pima-indians-diabetes.csv")
pd.set_option('display.max_columns', None)
print(train.head())
#print("Train :", train.shape)
#train.info()
#print(train.describe())
NaN_col_names = ['Plasma_glucose_concentration','blood_pressure','Triceps_skin_fold_thickness','serum_insulin','BMI']
#print((train[NaN_col_names] == 0).sum())

week5_drawFigure.drawFigure(train)
plt.show()
week5_FEtrain_lr.FEtrain_lr(train, 1)
week5_FELog_lr.FELog_lr(train, 1)